package com.gao.transportapi.common.convention.result;

import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import org.springframework.beans.BeanUtils;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.function.Function;
import java.util.stream.Collectors;

@Data
@NoArgsConstructor
@AllArgsConstructor
public class PageDTO<V> {
    private Long total;
    private Long pages;
    private List<V> list;
    
    /**
     * 用于缓存实例化的对象，提高性能
     */
    private static final Map<Class<?>, Function<Object, Object>> CONVERTOR_CACHE = new ConcurrentHashMap<>();
    
    /**
     * 数据量阈值，超过此值使用并行流处理
     */
    private static final int PARALLEL_THRESHOLD = 500;

    /**
     * 返回空分页结果
     * @param p MybatisPlus的分页结果
     * @param <V> 目标VO类型
     * @param <P> 原始PO类型
     * @return VO的分页对象
     */
    public static <V, P> PageDTO<V> empty(Page<P> p){
        return new PageDTO<>(p.getTotal(), p.getPages(), Collections.emptyList());
    }

    /**
     * 将MybatisPlus分页结果转为 VO分页结果
     * @param p MybatisPlus的分页结果
     * @param voClass 目标VO类型的字节码
     * @param <V> 目标VO类型
     * @param <P> 原始PO类型
     * @return VO的分页对象
     */
    public static <V, P> PageDTO<V> of(Page<P> p, Class<V> voClass) {
        // 1.非空校验
        List<P> records = p.getRecords();
        if (records == null || records.isEmpty()) {
            // 无数据，返回空结果
            return empty(p);
        }
        // 2.数据转换
        List<V> vos = copyToList(records, voClass);
        // 3.封装返回
        return new PageDTO<>(p.getTotal(), p.getPages(), vos);
    }

    /**
     * 将MybatisPlus分页结果转为 VO分页结果，允许用户自定义PO到VO的转换方式
     * @param p MybatisPlus的分页结果
     * @param convertor PO到VO的转换函数
     * @param <V> 目标VO类型
     * @param <P> 原始PO类型
     * @return VO的分页对象
     */
    public static <V, P> PageDTO<V> of(Page<P> p, Function<P, V> convertor) {
        // 1.非空校验
        List<P> records = p.getRecords();
        if (records == null || records.isEmpty()) {
            // 无数据，返回空结果
            return empty(p);
        }
        // 2.数据转换
        int size = records.size();
        List<V> vos;
        if (size <= PARALLEL_THRESHOLD) {
            vos = records.stream().map(convertor).collect(Collectors.toCollection(() -> new ArrayList<>(size)));
        } else {
            vos = records.parallelStream().map(convertor).collect(Collectors.toCollection(() -> new ArrayList<>(size)));
        }
        // 3.封装返回
        return new PageDTO<>(p.getTotal(), p.getPages(), vos);
    }

    /**
     * 将PO列表拷贝为VO列表
     * @param records PO列表
     * @param voClass VO类的字节码
     * @param <V> 目标VO类型
     * @param <P> 原始PO类型
     * @return VO列表
     */
    @SuppressWarnings("unchecked")
    private static <V, P> List<V> copyToList(List<P> records, Class<V> voClass) {
        int size = records.size();
        if (size == 0) {
            return Collections.emptyList();
        }
        
        // 获取或创建转换器
        Function<P, V> convertor = (Function<P, V>) CONVERTOR_CACHE.computeIfAbsent(
            voClass,
            clazz -> (Function<Object, Object>) source -> {
                Object target = BeanUtils.instantiateClass((Class<?>) clazz);
                BeanUtils.copyProperties(source, target);
                return target;
            }
        );
        
        // 根据数据量选择处理方式
        if (size <= PARALLEL_THRESHOLD) {
            // 小数据量使用普通流处理
        return records.stream()
                    .map(convertor)
                    .collect(Collectors.toCollection(() -> new ArrayList<>(size)));
        } else {
            // 大数据量使用并行流提高性能
            return records.parallelStream()
                    .map(convertor)
                    .collect(Collectors.toCollection(() -> new ArrayList<>(size)));
    }
    }
}